package demo;

/**
 * Created with IntelliJ IDEA.
 * Description :
 * User: jiume
 * Date: 2024-12-23
 * Time: 9:14
 */
public class Demo {
}


class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode() {
    }

    TreeNode(int val) {
        this.val = val;
    }

    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

class 计算布尔二叉树的值 {
    // 计算出root指针指向的树的布尔值
    public boolean evaluateTree(TreeNode root) {
        if (root.left == null && root.right == null) { // 叶子结点
            return root.val == 1 ? true : false;
        }
        // 先得计算出左子树
        boolean left = evaluateTree(root.left);
        // 再去计算右子树
        boolean right = evaluateTree(root.right);
        // 再根据根结点的值来计算出最终结果
        if (root.val == 2) { // or
            return left || right;
        } else { // and
            return left && right;
        }
    }
}

class 求根结点到叶子结点的数字之和 {
    public int sumNumbers(TreeNode root) {
        return dfs(root, 0);
    }

    private int dfs(TreeNode root, int prevSum) {
        // 把当前路径的值计算出来
        prevSum = prevSum * 10 + root.val;
        // 判断当前结点是否为叶子结点
        if (root.left == null && root.right == null) {
            return prevSum;
        }
        // 可能会存在左子树为空，右子树不为空；左子树不为空，右子树为空的情况
        // 这里没有去判断当前结点是否为null的情况，可能会出现空指针异常
        int sum = 0;
        if (root.left != null) {
            sum += dfs(root.left, prevSum);
        }
        if (root.right != null) {
            sum += dfs(root.right, prevSum);
        }
        return sum;
    }
}